"""
2044. 统计按位或能得到最大值的子集数目
中等
相关标签
premium lock icon
相关企业
提示
给你一个整数数组 nums ，请你找出 nums 子集 按位或 可能得到的 最大值 ，并返回按位或能得到最大值的 不同非空子集的数目 。

如果数组 a 可以由数组 b 删除一些元素（或不删除）得到，则认为数组 a 是数组 b 的一个 子集 。如果选中的元素下标位置不一样，则认为两个子集 不同 。

对数组 a 执行 按位或 ，结果等于 a[0] OR a[1] OR ... OR a[a.length - 1]（下标从 0 开始）。



示例 1：

输入：nums = [3,1]
输出：2
解释：子集按位或能得到的最大值是 3 。有 2 个子集按位或可以得到 3 ：
- [3]
- [3,1]
示例 2：

输入：nums = [2,2,2]
输出：7
解释：[2,2,2] 的所有非空子集的按位或都可以得到 2 。总共有 23 - 1 = 7 个子集。
示例 3：

输入：nums = [3,2,1,5]
输出：6
解释：子集按位或可能的最大值是 7 。有 6 个子集按位或可以得到 7 ：
- [3,5]
- [3,1,5]
- [3,2,5]
- [3,2,1,5]
- [2,5]
- [2,1,5]


提示：

1 <= nums.length <= 16
1 <= nums[i] <= 105
"""
from typing import List


class Solution:
    def countMaxOrSubsets(self, nums: List[int]) -> int:
        """
        选或不选两种，每种选择后结果次数+1
        :param nums:
        :return:
        """
        dp = {}
        for i in nums:
            # 收集当前需要处理的项，遍历过程中直接修改数据会有问题
            items_to_process = list(dp.items())
            for k, v in items_to_process:
                n = k | i
                dp[n] = dp.get(n, 0) + v
            # 使用get方法的简洁写法
            dp[i] = dp.get(i, 0) + 1
        max1 = 0
        for i in dp.keys():
            max1 = max(max1,i)
        return dp[max1]

if __name__ == '__main__':
    print(Solution().countMaxOrSubsets([3,1]))
    print(Solution().countMaxOrSubsets([2,2,2]))
    print(Solution().countMaxOrSubsets([3,2,1,5]))
